import fasttext

import numpy as np

# 加载训练好的 FastText 模型
model = fasttext.load_model('static/model_01/fasttext_model.bin')


def predict_probability(input_text):

    # 使用 fasttext_model 模型进行预测
    labels, scores = model.predict(input_text, k=2)

    # 将得分转换为概率值
    probabilities = np.exp(scores) / np.sum(np.exp(scores))

    # 假设你想获取 label 为 '__label__1' 的概率值
    label_1_index = labels.index('__label__1')
    label_1_prob = probabilities[label_1_index]

    return label_1_prob
